package liming.ai.data.deepseek;

import org.json.JSONObject;

// 消息实体类
public class Message {
    private final Role role;
    private final String content;

    public Message(Role role, String content) {
        this.role = role;
        this.content = content;
    }

    public JSONObject toJson() {
        JSONObject json = new JSONObject();
        json.put("role", role.name().toLowerCase());
        json.put("content", content);
        return json;
    }

    public enum Role {
        USER, ASSISTANT, SYSTEM
    }
}